pythonCopy code
from PIL import Image, ImageDraw, ImageFont

# Step 1: Open the image
image_path = "/mnt/data/IMG_0029.jpg"
image = Image.open(image_path)

# Step 2: Initialize the drawing object
draw = ImageDraw.Draw(image)

# Step 3: Load the font
font_path = "/mnt/data/NotoSansKR-Bold.ttf"
font_size = 100
font = ImageFont.truetype(font_path, font_size)

# Step 4: Specify the text and position
text = "Hello World"
text_position = (100, 100)

# Step 5: Add the text to image
draw.text(text_position, text, font=font, fill="white")

# Step 6: Save the new image
output_image_path = "/mnt/data/IMG_0029_generated.jpg"
image.save(output_image_path)

output_image_path

pythonCopy code
from PIL import Image
import IPython.display as display

# Display the generated image with text
display.display(Image.open(output_image_path))

pythonCopy code
# Create the full Python script that accepts command line arguments and writes text on the image

script_content = '''
import sys
from PIL import Image, ImageDraw, ImageFont

def write_text_on_image(text, font_path, font_size, image_path, output_image_path, auto_newline, h_align, v_align):
    # Open the image
    image = Image.open(image_path)
    draw = ImageDraw.Draw(image)
    
    # Load the font
    font = ImageFont.truetype(font_path, int(font_size))
    
    # Calculate text size and position
    text_width, text_height = draw.textsize(text, font=font)
    image_width, image_height = image.size
    
    if auto_newline == 1:
        # Add new lines if the text width is greater than image width
        lines = []
        words = text.split()
        while words:
            line = ''
            while words and draw.textsize(line + words[0], font=font)[0] <= image_width:
                line += (words.pop(0) + ' ')
            lines.append(line)
        
        text_height = len(lines) * (font_size + 5)
        text = '\n'.join(lines)
    
    # Horizontal alignment
    if h_align == 0:
        x = 0
    elif h_align == 1:
        x = (image_width - text_width) // 2
    else:
        x = image_width - text_width

    # Vertical alignment
    if v_align == 0:
        y = 0
    elif v_align == 1:
        y = (image_height - text_height) // 2
    else:
        y = image_height - text_height
    
    # Draw text on image
    draw.multiline_text((x, y), text, font=font, fill="white")
    
    # Save the image
    image.save(output_image_path)

if __name__ == "__main__":
    if len(sys.argv) != 9:
        print("Usage: python3 writeTextOnImage.py <text> <font_path> <font_size> <image_path> <output_image_path> <auto_newline> <h_align> <v_align>")
        sys.exit(1)
    
    text = sys.argv[1]
    font_path = sys.argv[2]
    font_size = sys.argv[3].replace("px", "")
    image_path = sys.argv[4]
    output_image_path = sys.argv[5]
    auto_newline = int(sys.argv[6])
    h_align = int(sys.argv[7])
    v_align = int(sys.argv[8])
    
    write_text_on_image(text, font_path, int(font_size), image_path, output_image_path, auto_newline, h_align, v_align)
'''

# Save the script as a Python file
script_file_path = '/mnt/data/writeTextOnImage.py'
with open(script_file_path, 'w') as f:
    f.write(script_content)

script_file_path

pythonCopy code
# Test the generated Python script using the subprocess module
import subprocess

# Prepare the command line arguments
text = "Hello World"
font_path = "/mnt/data/NotoSansKR-Bold.ttf"
font_size = "100px"
image_path = "/mnt/data/IMG_0029.jpg"
output_image_path = "/mnt/data/IMG_0029_generated_test.jpg"
auto_newline = "1"
h_align = "1"
v_align = "1"

# Execute the command
subprocess.run(["python3", script_file_path, text, font_path, font_size, image_path, output_image_path, auto_newline, h_align, v_align])

# Display the generated image to check if the text has been written correctly
display.display(Image.open(output_image_path))

pythonCopy code
# Correct the script content and save it again
# The script content had an un-terminated string. Fixing it.

# Save the corrected script as a Python file
script_file_path_corrected = '/mnt/data/writeTextOnImage_corrected.py'
with open(script_file_path_corrected, 'w') as f:
    f.write(script_content)

# Re-run the test
subprocess.run(["python3", script_file_path_corrected, text, font_path, font_size, image_path, output_image_path, auto_newline, h_align, v_align])

# Display the generated image to check if the text has been written correctly
display.display(Image.open(output_image_path))

pythonCopy code
# Fix the script content and save it again
# It seems that the script content itself has not been fixed properly.

# Re-check the script content
print(script_content[:500])  # Display first 500 characters to check for issues

# Looks like the script content was truncated in the first place. We'll regenerate it.
# Generate the corrected script content
script_content_corrected = script_content

# Save the corrected script as a Python file
script_file_path_corrected = '/mnt/data/writeTextOnImage_corrected.py'
with open(script_file_path_corrected, 'w') as f:
    f.write(script_content_corrected)

# Re-run the test
subprocess.run(["python3", script_file_path_corrected, text, font_path, font_size, image_path, output_image_path, auto_newline, h_align, v_align])

# Display the generated image to check if the text has been written correctly
display.display(Image.open(output_image_path))

pythonCopy code
# Let's attempt to write the script content again and save it to a new file.

# This time we will skip the 'script_content' variable and write directly to the file.

script_file_path_corrected = '/mnt/data/writeTextOnImage_corrected.py'

with open(script_file_path_corrected, 'w') as f:
    f.write('''
import sys
from PIL import Image, ImageDraw, ImageFont

def write_text_on_image(text, font_path, font_size, image_path, output_image_path, auto_newline, h_align, v_align):
    # Open the image
    image = Image.open(image_path)
    draw = ImageDraw.Draw(image)
    
    # Load the font
    font = ImageFont.truetype(font_path, int(font_size))
    
    # Calculate text size and position
    text_width, text_height = draw.textsize(text, font=font)
    image_width, image_height = image.size
    
    if auto_newline == 1:
        # Add new lines if the text width is greater than image width
        lines = []
        words = text.split()
        while words:
            line = ''
            while words and draw.textsize(line + words[0], font=font)[0] <= image_width:
                line += (words.pop(0) + ' ')
            lines.append(line)
        
        text_height = len(lines) * (font_size + 5)
        text = '\\n'.join(lines)
    
    # Horizontal alignment
    if h_align == 0:
        x = 0
    elif h_align == 1:
        x = (image_width - text_width) // 2
    else:
        x = image_width - text_width

    # Vertical alignment
    if v_align == 0:
        y = 0
    elif v_align == 1:
        y = (image_height - text_height) // 2
    else:
        y = image_height - text_height
    
    # Draw text on image
    draw.multiline_text((x, y), text, font=font, fill="white")
    
    # Save the image
    image.save(output_image_path)

if __name__ == "__main__":
    if len(sys.argv) != 9:
        print("Usage: python3 writeTextOnImage.py <text> <font_path> <font_size> <image_path> <output_image_path> <auto_newline> <h_align> <v_align>")
        sys.exit(1)
    
    text = sys.argv[1]
    font_path = sys.argv[2]
    font_size = sys.argv[3].replace("px", "")
    image_path = sys.argv[4]
    output_image_path = sys.argv[5]
    auto_newline = int(sys.argv[6])
    h_align = int(sys.argv[7])
    v_align = int(sys.argv[8])
    
    write_text_on_image(text, font_path, int(font_size), image_path, output_image_path, auto_newline, h_align, v_align)
''')

# Re-run the test
subprocess.run(["python3", script_file_path_corrected, text, font_path, font_size, image_path, output_image_path, auto_newline, h_align, v_align])

# Display the generated image to check if the text has been written correctly
display.display(Image.open(output_image_path))

pythonCopy code
# Update the script to handle text wrapping correctly when the text width exceeds the image width

updated_script_content = '''
import sys
from PIL import Image, ImageDraw, ImageFont

def write_text_on_image(text, font_path, font_size, image_path, output_image_path, auto_newline, h_align, v_align):
    # Open the image
    image = Image.open(image_path)
    draw = ImageDraw.Draw(image)
    
    # Load the font
    font = ImageFont.truetype(font_path, int(font_size))
    
    # Calculate text size and position
    image_width, image_height = image.size

    if auto_newline == 1:
        # Add new lines if the text width is greater than image width
        lines = []
        words = text.split()
        while words:
            line = ''
            while words and draw.textsize(line + words[0], font=font)[0] <= image_width:
                line += (words.pop(0) + ' ')
            lines.append(line.strip())
        
        text_height = len(lines) * (font.getsize('A')[1] + 5)
        text = '\\n'.join(lines)
    
    text_width, text_height = draw.textsize(text, font=font, spacing=5)
    
    # Horizontal alignment
    if h_align == 0:
        x = 0
    elif h_align == 1:
        x = (image_width - text_width) // 2
    else:
        x = image_width - text_width

    # Vertical alignment
    if v_align == 0:
        y = 0
    elif v_align == 1:
        y = (image_height - text_height) // 2
    else:
        y = image_height - text_height
    
    # Draw text on image
    draw.multiline_text((x, y), text, font=font, fill="white", spacing=5)
    
    # Save the image
    image.save(output_image_path)

if __name__ == "__main__":
    if len(sys.argv) != 9:
        print("Usage: python3 writeTextOnImage.py <text> <font_path> <font_size> <image_path> <output_image_path> <auto_newline> <h_align> <v_align>")
        sys.exit(1)
    
    text = sys.argv[1]
    font_path = sys.argv[2]
    font_size = sys.argv[3].replace("px", "")
    image_path = sys.argv[4]
    output_image_path = sys.argv[5]
    auto_newline = int(sys.argv[6])
    h_align = int(sys.argv[7])
    v_align = int(sys.argv[8])
    
    write_text_on_image(text, font_path, int(font_size), image_path, output_image_path, auto_newline, h_align, v_align)
'''

# Save the updated script as a Python file
updated_script_file_path = '/mnt/data/writeTextOnImage_updated.py'
with open(updated_script_file_path, 'w') as f:
    f.write(updated_script_content)

# Re-run the test with a longer text to verify the auto_newline functionality
long_text = "Hello World! " * 10
subprocess.run(["python3", updated_script_file_path, long_text, font_path, font_size, image_path, output_image_path, "1", "1", "1"])

# Display the generated image to check if the text has been written correctly
display.display(Image.open(output_image_path))